from transformers import AutoModel
model = AutoModel.from_pretrained("/home/linweibin/lwb2/model/checkpoint_0.ckpt")